'''
Author: zhao-leo 18055219130@163.com
Date: 2024-10-22 21:37:20
LastEditTime: 2024-10-22 23:01:45
'''
from src.download_models.downloader import modelDownloader
from src.torch_use.bert import bertModel
from src.torch_use.squad2 import Squad2

CONFIG_PATH = 'models.toml'
modelDownloader(CONFIG_PATH).load_models()
model1 = bertModel("./.cache/models/bert-base-chinese/")
model1.run_model(show_result=True)
print("--------------------------------------------------------------------\n")
model2 = Squad2("./.cache/models/mdeberta-v3-base-squad2/")
model2.run_model()